leetcode 2616 problem solution in python